lightning
pillow
torchvision
imgaug
matplotlib